import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

/**
 * 双色球中奖等级判断工具类
 */
public class LotteryChecker {

    /**
     * 判断中奖等级
     * @param userRedBalls 用户投注的6个红球（无序）
     * @param userBlueBall 用户投注的1个蓝球
     * @param drawRedBalls 开奖的6个红球（无序）
     * @param drawBlueBall 开奖的1个蓝球
     * @return 中奖等级（1-6等，0为未中奖）
     */
    public static int checkPrizeLevel(int[] userRedBalls, int userBlueBall,
                                      int[] drawRedBalls, int drawBlueBall) {
        // 统计红球匹配数
        int redMatchCount = countRedMatch(userRedBalls, drawRedBalls);
        // 蓝球是否匹配
        boolean blueMatch = (userBlueBall == drawBlueBall);


        // 按规则判断等级
        if (redMatchCount == 6 && blueMatch) {
            return 1; // 一等奖
        } else if (redMatchCount == 6 && !blueMatch) {
            return 2; // 二等奖
        } else if (redMatchCount == 5 && blueMatch) {
            return 3; // 三等奖
        } else if (redMatchCount == 5 || (redMatchCount == 4 && blueMatch)) {
            return 4; // 四等奖
        } else if (redMatchCount == 4 || (redMatchCount == 3 && blueMatch)) {
            return 5; // 五等奖
        } else if (blueMatch) {
            return 6; // 六等奖
        } else {
            return 0; // 未中奖
        }
    }


    /**
     * 统计用户红球与开奖红球的匹配数量
     * @param userRed  用户红球数组
     * @param drawRed 开奖红球数组
     * @return 匹配数量
     */
    private static int countRedMatch(int[] userRed, int[] drawRed) {
        Set<Integer> drawRedSet = new HashSet<>();
        for (int num : drawRed) {
            drawRedSet.add(num);
        }

        int count = 0;
        for (int num : userRed) {
            if (drawRedSet.contains(num)) {
                count++;
            }
        }
        return count;
    }


    /**
     * 校验用户输入的号码是否合法
     * @param userNumbers 用户输入的7个号码（前6红+1蓝）
     * @return 合法返回true，否则false
     */
    public static boolean validateUserInput(int[] userNumbers) {
        // 长度必须是7
        if (userNumbers.length != 7) {
            return false;
        }

        // 校验红球（前6个）：1-33，不重复
        Set<Integer> redSet = new HashSet<>();
        for (int i = 0; i < 6; i++) {
            int red = userNumbers[i];
            if (red < 1 || red > 33) {
                return false;
            }
            if (!redSet.add(red)) { // 重复
                return false;
            }
        }

        // 校验蓝球（第7个）：1-16
        int blue = userNumbers[6];
        return blue >= 1 && blue <= 16;
    }
}
